package com.ruoyi.home.jinrong.utils;

import com.ruoyi.common.core.constant.CacheConstants;
import org.apache.commons.collections4.Get;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.VerticalAlignment;

/**
 * @author 郭豪
 * @date 2021/8/2 9:57
 * @description:
 */
public class ExcelUtils {
    /**
     *
     * @param sheetName sheet名称
     * @param title      标题
     * @param values     内容
     * @return
     */
    public static HSSFWorkbook getExcel(String sheetName , String[] title , String[][] values){
        // 第一步，创建一个webbook，对应一个Excel文件
        HSSFWorkbook wb = new HSSFWorkbook();

        // 第二步，在webbook中添加一个sheet,对应Excel文件中的sheet
        HSSFSheet sheet = wb.createSheet("交易流水表");

        // 第三步，在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
        HSSFRow row = sheet.createRow((int) 0);

        // 第四步，创建单元格，并设置值表头 设置表头居中
        HSSFCellStyle style = wb.createCellStyle();

        //style.setAlignment(HSSFCellStyle.ALIGN_CENTER); // 创建一个居中格式
        style.setVerticalAlignment(VerticalAlignment.CENTER);

        //声明一个列对象
        HSSFCell cell = null;

        //创建标题
        for (int i = 0; i < title.length; i++) {
            //列头
            cell = row.createCell(i);
            //设置格式
            cell.setCellStyle(style);
            //写入标题
            cell.setCellValue(title[i]);
        }

        //写入内容
        for (int i = 0; i < values.length; i++) {
            //获取对应的行对象，从第一行开始
            row = sheet.createRow(i+1);
            for (int j = 0; j < values[i].length; j++) {
                //将内容按顺序赋给对应的列对象ss
                row.createCell(j).setCellValue(values[i][j]);
            }
        }
        return wb;
    }
}
